from torch.optim import Adam  # noqa